#pragma once

#include <vector>
#include <stdint.h>

class OneHot
{
public:
    static std::vector<std::vector<uint8_t>> encode(const std::vector<uint8_t> &labels, uint64_t num_classes = 10);
    static std::vector<uint8_t> decode(const std::vector<std::vector<uint8_t>> &one_hot_labels);
};
